update nvlink checks#143

Open

ncclementi wants to merge 1 commit intorapidsai:mainfrom

ncclementi:update-nvlink-check

ncclementi commented Mar 30, 2026

Closes #142

This PR (mostly made with Claude):

makes sure that we don't raise when NVLink it's not supported,
makes sure that we don't only check NVLink for device 0.
Has better raises, to give more informative information to the user when the check fails.
Updated the tests to make sure we check all the different scenarios.


          update nvlink checks

b44d8fd

ncclementi requested a review from a team as a code owner

March 30, 2026 20:31

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet